resnet_distributed.py